doing graph theory with python